package net.lmc.realtime.jtp.dws.log.function;

import net.lmc.realtime.jtp.dws.log.bean.PageViewBean;
import org.apache.commons.lang3.time.FastDateFormat;
import org.apache.flink.streaming.api.functions.windowing.WindowFunction;
import org.apache.flink.streaming.api.windowing.windows.TimeWindow;
import org.apache.flink.util.Collector;

public class PageViewWindowFunction implements WindowFunction<PageViewBean,String,String, TimeWindow> {

    private FastDateFormat format = FastDateFormat.getInstance("yyyy-MM-dd HH:mm:ss");

    @Override
    public void apply(String s, TimeWindow window, Iterable<PageViewBean> input, Collector<String> out) throws Exception {

        //获取开始和结束时间
        String windowStartTime = format.format(window.getStart());
        String windowEndTime = format.format(window.getEnd());

        //进行数据计算
        Long pvCount = 0L;
        Long pvDuringTime = 0L;
        Long uvCount = 0L;
        Long sessionCount = 0L;
        for (PageViewBean bean : input){
            pvCount += bean.getPageCount();
            pvDuringTime += bean.getPvDuringTime();
            uvCount += bean.getUvCount();
            sessionCount += bean.getSessionCount();
        }

        //输出数据
        String ss = windowStartTime + "," + windowEndTime + "," + s + "," + pvCount + ","
                + pvDuringTime + "," + uvCount + "," + sessionCount + "," + System.currentTimeMillis();

        out.collect(ss);
    }
}
